#pragma once
#include "PnnLABQuantizer.h"
#include "APNsgaIII.h"

#include <string>

namespace PnnLABQuant
{
	// =============================================================
	// Quantizer objects and functions
	//
	// COVERED CODE IS PROVIDED UNDER THIS LICENSE ON AN "AS IS" BASIS, WITHOUT WARRANTY
	// OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, WITHOUT LIMITATION, WARRANTIES
	// THAT THE COVERED CODE IS FREE OF DEFECTS, MERCHANTABLE, FIT FOR A PARTICULAR PURPOSE
	// OR NON-INFRINGING. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE COVERED
	// CODE IS WITH YOU. SHOULD ANY COVERED CODE PROVE DEFECTIVE IN ANY RESPECT, YOU (NOT
	// THE INITIAL DEVELOPER OR ANY OTHER CONTRIBUTOR) ASSUME THE COST OF ANY NECESSARY
	// SERVICING, REPAIR OR CORRECTION. THIS DISCLAIMER OF WARRANTY CONSTITUTES AN ESSENTIAL
	// PART OF THIS LICENSE. NO USE OF ANY COVERED CODE IS AUTHORIZED HEREUNDER EXCEPT UNDER
	// THIS DISCLAIMER.
	//
	// Use at your own risk!
	// =============================================================

	class PnnLABGAQuantizer
	{
	private:
		//Asserts floating point compatibility at compile time
		static_assert(std::numeric_limits<float>::is_iec559, "IEEE 754 required");

		double _fitness = -numeric_limits<double>::infinity();
		double _ratioX = 0, _ratioY = 0;
		vector<double> _convertedObjectives;
		vector<double> _objectives;
		vector<vector<ARGB> > m_pixelsList;
		unique_ptr<PnnLABQuantizer> m_pq;

		void calculateError(vector<double>& errors);
		void calculateFitness();
		string getRatioKey() const;
		auto findByRatioKey(const string& ratioKey) const;
		void clear();

	public:
		PnnLABGAQuantizer(PnnLABQuantizer& pq, const vector<shared_ptr<Bitmap> >& pSources, UINT nMaxColors);
		PnnLABGAQuantizer(PnnLABQuantizer& pq, const vector<vector<ARGB> >& pixelsList, const vector<UINT>& bitmapWidths, UINT nMaxColors);

		float getFitness();
		shared_ptr<PnnLABGAQuantizer> crossover(const PnnLABGAQuantizer& mother, int numberOfCrossoverPoints, float crossoverProbability);
		bool dominates(const PnnLABGAQuantizer* right);
		void mutation(int mutationSize, float mutationProbability);
		vector<double> getObjectives() const;
		vector<double>& getConvertedObjectives();
		void resizeConvertedObjectives(int numObj);
		shared_ptr<PnnLABGAQuantizer> makeNewFromPrototype();

		UINT getMaxColors() const;
		string getResult() const;
		bool hasAlpha() const {
			return m_pq->hasAlpha();
		}
		void setRatio(double ratioX, double ratioY);
		bool QuantizeImage(vector<shared_ptr<Bitmap> >& pBitmaps, bool dither = true);
	};

}
